#include "slist.h"
#include <stdio.h>
#include <string.h>
#include <stdlib.h>

// 节点的创建
static node_t *__node_create(slist_t *s, void *data)
{
    node_t *tmp = NULL;

    // 参数检查
    if (NULL == s || NULL == data)
    {
        perror("argument lawless...");
        goto err0;
    }

    // 创建节点空间
    tmp = malloc(sizeof(node_t));
    if (NULL == tmp)
    {
        perror("tmp malloc failed...");
        goto err0;
    }
    memset(tmp, 0, sizeof(node_t));

    // 创建节点内变量的空间
    tmp->data = malloc(s->size);
    if (NULL == tmp->data)
    {
        perror("tmp_data malloc failed...");
        goto err1;
    }
    memset(tmp->data, 0, s->size);

    // 节点赋值
    memcpy(tmp->data, data, s->size);
    tmp->next = NULL;

    // 增加节点个数
    s->count++;

    return tmp;



err1:
    free(tmp);
    tmp = NULL;
err0:
    return NULL;
}

// 头信息结构体创建
int slist_create(slist_t **s, int size, ft1 my_destroy)
{
    // 参数检查
    if (size <= 0 || NULL == my_destroy)
    {
        perror("argument lawless...");
        goto err0;
    }

    // 创建结构体
    *s = malloc(sizeof(slist_t));
    if (NULL == *s)
    {
        perror("*s malloc failed...");
        goto err0;
    }
    memset(*s, 0, sizeof(slist_t));

    // 赋值
    (*s)->count = 0;
    (*s)->my_destroy = my_destroy;
    (*s)->p = NULL;
    (*s)->size = size;


    return 0;
err0:
    return -1;
}

// 链表尾部插入
int slist_append(slist_t *s, void *data)
{
    node_t *new_node = NULL;
    node_t *tmp = NULL;

    // 参数检查
    if (NULL == s || NULL == data)
    {
        perror("argument lawless...");
        goto err0;
    }

    // 创建节点
    new_node = __node_create(s, data);

    // 尾部插入链表
    tmp = s->p;
    if (tmp == NULL)
    {
        s->p = new_node;
    }
    else 
    {
        while (NULL != tmp->next)
        {
            tmp = tmp->next;
        }
        tmp->next = new_node;
    }


    return 0;
err0:
    return -1;
}

// 链表头部插入
int slist_prepend(slist_t *s, void *data)
{
    node_t *new_node = NULL;

    // 参数检查
    if (NULL == s || NULL == data)
    {
        perror("argument lawless...");
        goto err0;
    }

    // 创建节点
    new_node = __node_create(s, data);

    // 链表头部插入
    new_node->next = s->p;
    s->p = new_node;



    return 0;
err0:
    return -1;
}

// 链表遍历
int slist_traverse(slist_t *s, ft1 my_print)
{
    node_t *node = NULL;
    node_t *tmp = NULL;

    // 参数检查
    if (NULL == s || NULL == my_print)
    {
        perror("argument lawless...");
        goto err0;
    }

    // 链表遍历
    tmp = s->p;
    while (NULL != tmp)
    {
        my_print(tmp);
        tmp = tmp->next;
    }
    printf("\n");

    return 0;
err0:
    return -1;
}

// 链表的销毁
int slist_destroy(slist_t *s)
{
    node_t *tmp = NULL;
    node_t *save = NULL;

    // 参数检查
    if (NULL == s)
    {
        perror("argument lawless...");
        goto err0;
    }

    tmp = s->p;
    // 节点的释放
    while (NULL != tmp)
    {
        // 保存下一个节点位置
        save = tmp->next;

        // 销毁节点
        s->my_destroy(tmp);
        s->count--;


        // 指向下一个节点
        tmp = save;
    }

    s->p = NULL;


    return 0;

err0:
    return -1;
}


// 检索插入
int slist_insert_by_index(slist_t *s, void *data, int index)
{
    node_t *new_node = NULL;
    int idx = index - 1;
    int i = 0;
    node_t *tmp = NULL;

    // 参数检查
    if (NULL == s || NULL == data)
    {
        perror("argument lawless...");
        goto err0;
    }

    // 检索插入
    if (idx < 0)
    {
        perror("argument lawless...");
        goto err0;
    }
    else if (idx == 0)
    {
        slist_prepend(s, data);
    }
    else if (idx >= s->count)
    {
        slist_append(s, data);
    }
    else
    {
        // 创建新的节点
        new_node = __node_create(s, data);

        // 查找检索位置
        tmp = s->p;
        for (i = 0; i < idx - 1; i++)
        {
            tmp = tmp->next;
        }

        // 插入
        new_node->next = tmp->next;
        tmp->next = new_node;

    }


    return 0;
err0:
    return -1;
}

// 检索删除
int slist_delete_by_index(slist_t *s, int index)
{
    node_t *tmp = NULL;
    node_t *save = NULL;
    int idx = index - 1;
    int i = 0;

    // 参数检查
    if (NULL == s)
    {
        perror("argument lawless...");
        goto err0;
    }

    if (index < 1 || index > s->count)
    {
        perror("argument lawless...");
        goto err0;
    }
    else if (s->count == 1)
    {
        s->my_destroy(s->p);
        s->p = NULL;
    }
    else if (index == 1)
    {
        // 保存
        save = s->p->next;

        // 释放空间
        s->my_destroy(s->p);

        // 指针设置
        s->p = save;
    }
    else
    {
        tmp = s->p;
        for (i = 0; i < idx - 1; i++)
        {
            tmp = tmp->next;
        }
        save = tmp->next->next;
        s->my_destroy(tmp->next);
        tmp->next = save;
    }

    // 节点个数-1
    s->count--;



    return 0;
err0:
    return -1;
}